import torch
from torch.utils.data import DataLoader,Dataset

class FeatureDataset(Dataset):

    def __init__(self,featurename,modality='train'):
        super(Dataset,self).__init__()
        self.modality = modality
        self.featurename = featurename
        if featurename not in ['tsn_score']:
            raise LookupError('{} not in given list'.format(featurename))
        if featurename == 'tsn_score':
            pass

    def __len__(self):
        pass

    def __index__(self):
        pass
